查看原文
其他

Github|教你用PyTorch打造专属GAN二次元头像(附源代码)

2017-03-08 全球人工智能 全球人工智能

全球人工智能


来源:Github Jie Le

翻译:孙菁


随机生成图像

通过DCGAN模型在包含143,000张动漫人脸的数据集上迭代训练100次,我们得到了本次实验所用的图像。



图像转换

通过执行隐含代码,使图像完成从第一列到最后一列的转换。



原始图像

我们可以观察到图像包含很多异常值,即有很多噪音,这些异常值会使得生成图像的质量有所下降。




使用

执行以下命令以启动整个程序:

$ python main.py --dataRoot path_to_dataset/


DCGAN的预训练模型呀包含在此目录中,使用jupyternotebook工具即可查看。


动漫人脸数据集

利用爬虫工具 gallery-dl,我们 从 danbooru.donmai.us中收集到了带有126个标记的动漫样式图片。之后,这些图片由动漫人脸识别器 python-animeface 处理。经过处理的数据集总共包含143,000张动漫人脸。需要注意的是,一些标记在经过剪裁后就没有意义了,比如,一张在“制服”标签下的图片在剪裁后就不再包含制服的部分了。


如何从头构件数据集?

预备工具:gallery-dl, python-animeface

1、下载动漫样式的图片



2、从下载的图片中提取动漫人脸



读者可以从build_face_dataset.py中寻找具体实现。

数据集(~400MB)也可以从此处下载(非商业使用)。


Github库点击阅读原文跳转


热门文章推荐

重磅|MIT发布脑控机器人:用脑电波(10毫秒分类)纠正机器人错误

重磅|谷歌预言:2029年通过纳米机器人和器官再造 或将实现人类永生

重磅|Messenger bot错误率高达70% Facebook被迫削减AI投资

招聘|腾讯大规模招聘AI开发工程师 年薪30-80W

讨论|周志华教授gcForest论文的价值与技术讨论(微信群)

最新|李飞飞:人口普查不用上门,谷歌街景加深度学习就搞定(附论文)

最新 | 百度最新“Deep Voice”语音技术 比WaveNet提速 400 倍(译)

重磅 |Boston Dynamics最新轮式暴跳机器人:身高1.98m 纵跳1.2m(译)

重磅 |“AI武器”已成特朗普媒体舆论首选工具!

技术 | 教你如何用“决策树算法”解决相亲问题?

您可能也对以下帖子感兴趣

文章有问题?点此查看未经处理的缓存